Tratamento dos dados
In [ ]:
Copied!
import numpy as np
import os
import zipfile
import cv2
import tensorflow as tf
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import warnings
import numpy as np
import os
import zipfile
import cv2
import tensorflow as tf
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import warnings
Armazenando todos os arquivos em uma lista
In [7]:
Copied!
diretorio = 'data/terra'
arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))]
print(arquivos)
diretorio = 'data/terra'
arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))]
print(arquivos)
['data/terra/-0.625°N, 102°E.jpg', 'data/terra/-1.75°N, 309.375°E.jpg', 'data/terra/-2°N, 305.125°E.jpg', 'data/terra/-2°N, 306.625°E.jpg', 'data/terra/-3.625°N, 15°E.jpg', 'data/terra/-3°N, 299.75°E.jpg', 'data/terra/-4.125°N, 286.75°E.jpg', 'data/terra/-5.125°N, 298.75°E.jpg', 'data/terra/-8.125°N, 295.875°E.jpg', 'data/terra/0.125°N, 113.5°E.jpg', 'data/terra/0.25°N, 109.375°E.jpg', 'data/terra/15°N, 270°E.jpg', 'data/terra/20.75°N, 283.625°E.jpg', 'data/terra/25°N, 281.375°E.jpg', 'data/terra/28.375°N, 98.75°E.jpg', 'data/terra/36.27°N, 138.57°E.jpg', 'data/terra/36.5°N, 139.5°E.jpg', 'data/terra/4.875°N, 304.75°E.jpg', 'data/terra/44.25°N, 281°E.jpg', 'data/terra/44.75°N, 273.125°E.jpg', 'data/terra/49.5°N, 139.625°E.jpg', 'data/terra/4°N, 286.75°E.jpg', 'data/terra/53.707°N, 96.848°E.jpg', 'data/terra/5°N, 18.625°E.jpg', 'data/terra/6°N, 359.875°E.jpg', 'data/terra/9.75°N, 38.625°E.jpg', 'data/terra/9°N, 349.25°E.jpg']
Verificando tamanho da imagem
In [14]:
Copied!
imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg')
imagem.shape
imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg')
imagem.shape
Out[14]:
(527, 1585, 3)
In [9]:
Copied!
largura, altura = 1585, 527
largura, altura = 1585, 527
Out[9]:
(527, 1585, 3)
Passando imagens de treino para cores de cinza
In [13]:
Copied!
imagens = []
classes = []
for imagem_caminho in arquivos:
try:
imagem = cv2.imread(imagem_caminho)
(H, W) = imagem.shape[:2]
except:
continue
imagem = cv2.resize(imagem, (largura, altura))
imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)
plt.imshow(imagem)
plt.show()
imagem = imagem.ravel()
imagens.append(imagem)
classes.append('terra')
imagens = []
classes = []
for imagem_caminho in arquivos:
try:
imagem = cv2.imread(imagem_caminho)
(H, W) = imagem.shape[:2]
except:
continue
imagem = cv2.resize(imagem, (largura, altura))
imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)
plt.imshow(imagem)
plt.show()
imagem = imagem.ravel()
imagens.append(imagem)
classes.append('terra')
In [ ]:
Copied!
In [ ]:
Copied!